-- [nfnl] fnl/conjure/client/lua/neovim.fnl
local _local_1_ = require("conjure.nfnl.module")
local autoload = _local_1_.autoload
local define = _local_1_.define
local core = autoload("conjure.nfnl.core")
local str = autoload("conjure.nfnl.string")
local config = autoload("conjure.config")
local mapping = autoload("conjure.mapping")
local log = autoload("conjure.log")
local fs = autoload("conjure.fs")
local extract = autoload("conjure.extract")
local M = define("conjure.client.lua.neovim", {["buf-suffix"] = ".lua", ["comment-prefix"] = "-- "})
M["form-node?"] = function(node)
  return (("function_call" == node:type()) or ("function_definition" == node:type()) or ("function_declaration" == node:type()) or ("local_declaration" == node:type()) or ("variable_declaration" == node:type()) or ("if_statement" == node:type()) or ("for_statement" == node:type()) or ("assignment_statement" == node:type()))
end
config.merge({client = {lua = {neovim = {persistent = "debug"}}}})
if config["get-in"]({"mapping", "enable_defaults"}) then
  config.merge({client = {lua = {neovim = {mapping = {reset_env = "rr", reset_all_envs = "ra"}}}}})
else
end
local cfg = config["get-in-fn"]({"client", "lua", "neovim"})
M.repls = (M.repls or {})
M["reset-env"] = function(filename)
  local filename0 = (filename or fs["localise-path"](extract["file-path"]()))
  M.repls[filename0] = nil
  return log.append({(M["comment-prefix"] .. "Reset environment for " .. filename0)}, {["break?"] = true})
end
M["reset-all-envs"] = function()
  local function _3_(filename)
    M.repls[filename] = nil
    return nil
  end
  core["run!"](_3_, core.keys(M.repls))
  return log.append({(M["comment-prefix"] .. "Reset all environments")}, {["break?"] = true})
end
M["on-filetype"] = function()
  local function _4_()
    return M["reset-env"]()
  end
  mapping.buf("LuaResetEnv", cfg({"mapping", "reset_env"}), _4_, {desc = "Reset the Lua REPL environment"})
  local function _5_()
    return M["reset-all-envs"]()
  end
  return mapping.buf("LuaResetAllEnvs", cfg({"mapping", "reset_all_envs"}), _5_, {desc = "Reset all Lua REPL environments"})
end
local function display(out, ret, err)
  local outs
  local function _6_(_241)
    return (M["comment-prefix"] .. "(out) " .. _241)
  end
  local function _7_(_241)
    return ("" ~= _241)
  end
  outs = core.map(_6_, core.filter(_7_, str.split((out or ""), "\n")))
  local errs
  local function _8_(_241)
    return (M["comment-prefix"] .. "(err) " .. _241)
  end
  local function _9_(_241)
    return ("" ~= _241)
  end
  errs = core.map(_8_, core.filter(_9_, str.split((err or ""), "\n")))
  log.append(outs)
  log.append(errs)
  return log.append(str.split(("res = " .. vim.inspect(ret)), "\n"))
end
local function lua_compile(opts)
  if (opts.origin == "file") then
    return loadfile(opts["file-path"])
  else
    local f, e = load(("return (" .. opts.code .. "\n)"))
    if f then
      return f, e
    else
      return load(opts.code)
    end
  end
end
M["default-env"] = function()
  local base = setmetatable({["REDIRECTED-OUTPUT"] = "", io = setmetatable({}, {__index = _G.io})}, {__index = _G})
  local print_redirected
  local function _12_(...)
    base["REDIRECTED-OUTPUT"] = (base["REDIRECTED-OUTPUT"] .. str.join("\t", {...}) .. "\n")
    return nil
  end
  print_redirected = _12_
  local io_write_redirected
  local function _13_(...)
    base["REDIRECTED-OUTPUT"] = (base["REDIRECTED-OUTPUT"] .. str.join({...}))
    return nil
  end
  io_write_redirected = _13_
  local io_read_redirected
  local function _14_()
    return ((extract.prompt("Input required: ") or "") .. "\n")
  end
  io_read_redirected = _14_
  base["print"] = print_redirected
  base.io["write"] = io_write_redirected
  base.io["read"] = io_read_redirected
  return base
end
local function pcall_default(f)
  local env = M["default-env"]()
  setfenv(f, env)
  local status, ret = pcall(f)
  return status, ret, env["REDIRECTED-OUTPUT"]
end
local function pcall_persistent_debug(file, f)
  M.repls[file] = (M.repls[file] or {})
  M.repls[file]["env"] = (M.repls[file].env or M["default-env"]())
  M.repls[file].env["REDIRECTED-OUTPUT"] = ""
  setfenv(f, M.repls[file].env)
  local collect_env
  local function _15_(_, _0)
    debug.sethook()
    local i = 1
    local n = true
    local v = nil
    while n do
      n, v = debug.getlocal(2, i)
      if n then
        M.repls[file].env[n] = v
        i = (i + 1)
      else
      end
    end
    return nil
  end
  collect_env = _15_
  debug.sethook(collect_env, "r")
  local status, ret = pcall(f)
  return status, ret, M.repls[file].env["REDIRECTED-OUTPUT"]
end
local function lua_eval(opts)
  local f, e = lua_compile(opts)
  if f then
    local pcall_custom
    do
      local case_17_ = cfg({"persistent"})
      if (case_17_ == "debug") then
        local partial_18_ = opts["file-path"]
        local function _19_(...)
          return pcall_persistent_debug(partial_18_, ...)
        end
        pcall_custom = _19_
      else
        local _ = case_17_
        pcall_custom = pcall_default
      end
    end
    local status, ret, out = pcall_custom(f)
    if status then
      return out, ret, ""
    else
      return out, nil, ("Execution error: " .. ret)
    end
  else
    return "", nil, ("Compilation error: " .. e)
  end
end
M["eval-str"] = function(opts)
  local out, ret, err = lua_eval(opts)
  display(out, ret, err)
  if opts["on-result"] then
    return opts["on-result"](vim.inspect(ret))
  else
    return nil
  end
end
M["eval-file"] = function(opts)
  M["reset-env"](opts["file-path"])
  local out, ret, err = lua_eval(opts)
  display(out, ret, err)
  if opts["on-result"] then
    return opts["on-result"](vim.inspect(ret))
  else
    return nil
  end
end
return M
